import matplotlib.pyplot as plt
import xlrd
from xlrd import open_workbook

# 定义list数组存储数据
xpix_data = []
x_data = []
y_data = []
z_data = []
huhu = []
wb = open_workbook(r'C:\Users\30348\Desktop\操作系统\Huhu_OS\huhu.xls')
for s in wb.sheets():
    x_data1 = list(range(1, s.nrows + 1))
    print(x_data1)
    # 读取一个工作簿中的每一列数据
    for col in range(s.ncols):
        # 统计整数坐标，设计数据的x轴坐标
        # print(col)
        xpix_data.append(col)

        # 定义values数组，存储读取的初始数据
        values = []
        # 读取每一列中的每行数据
        for row in range(s.nrows):
            # 将读取到的数据添加到values数组中
            values.append(s.cell(row, col).value)
        # print(values)
        huhu.append(values)
print(huhu[0])
print(huhu[1])
# print(huhu[2])
# 添加网格，方便观察数据图像
plt.grid()
# 添加坐标轴信息，和图例说明
for i in range(len(huhu)):
    plt.plot(x_data1, huhu[i], linewidth=i + 1, label='消费者' + str(i + 1))
    for x, y in zip(x_data1, huhu[i]):
        plt.text(x + 0.1, y + 1, str(int(y)), ha='center', va='bottom', fontsize=10.5)
# plt.plot(x_data1, huhu[0], label='x轴数据')
# plt.plot(x_data1, huhu[1], label='y轴数据')
# plt.plot(x_data1, huhu[2], label='z轴数据')
# print(s.nrows)

# 解决中文显示问题
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False

# 给图像添加必要的文字说明
plt.title(u"Compare_Sum")
plt.xlabel(u"Times")
plt.ylabel(u"Sum_Of_Consumer")
plt.legend()

plt.show()
print("over!")
